import { LucideIcon, Bot, Zap, Eye, Code, Search, Star } from 'lucide-react';

export interface AIModel {
  id: string;
  name: string;
  provider: string;
  providerLogo?: React.ReactNode;
  description: string;
  speed: number; // 1-5
  intelligence: number; // 1-5
  context: string;
  supports: {
    webSearch: boolean;
    vision: boolean;
    tools: boolean;
  };
  tags: string[];
}

export const models: AIModel[] = [
  {
    id: 'gemini-2.0-flash',
    name: 'Gemini 2.0 Flash',
    provider: 'Google',
    providerLogo: <Bot className="w-4 h-4" />,
    description: 'Low-latency model optimized for agentic experiences.',
    speed: 5,
    intelligence: 3,
    context: '1M tokens',
    supports: {
      webSearch: true,
      vision: true,
      tools: true,
    },
    tags: ['Recommended'],
  },
  {
    id: 'sonar',
    name: 'Sonar',
    provider: 'Perplexity',
    providerLogo: <Zap className="w-4 h-4" />,
    description: 'A fast and capable model for a wide range of tasks.',
    speed: 4,
    intelligence: 4,
    context: '32K tokens',
    supports: {
      webSearch: true,
      vision: false,
      tools: true,
    },
    tags: [],
  },
  {
    id: 'gpt-4o-mini',
    name: 'GPT-4o mini',
    provider: 'OpenAI',
    providerLogo: <Star className="w-4 h-4" />,
    description: 'OpenAI\'s latest, most advanced mini model.',
    speed: 4,
    intelligence: 5,
    context: '128K tokens',
    supports: {
      webSearch: true,
      vision: true,
      tools: true,
    },
    tags: ['Recommended'],
  },
  {
    id: 'claude-3.5-haiku',
    name: 'Claude 3.5 Haiku',
    provider: 'Anthropic',
    providerLogo: <Eye className="w-4 h-4" />,
    description: 'The fastest and most compact model in the Claude 3.5 family.',
    speed: 5,
    intelligence: 4,
    context: '200K tokens',
    supports: {
      webSearch: true,
      vision: true,
      tools: true,
    },
    tags: [],
  },
  {
    id: 'deepseek-v3',
    name: 'DeepSeek-V3',
    provider: 'Together AI',
    providerLogo: <Code className="w-4 h-4" />,
    description: 'A powerful open-source model specialized in coding.',
    speed: 3,
    intelligence: 4,
    context: '128K tokens',
    supports: {
      webSearch: false,
      vision: false,
      tools: true,
    },
    tags: [],
  },
  {
    id: 'llama-3.3-70b',
    name: 'Llama 3.3 70B',
    provider: 'Groq',
    providerLogo: <Search className="w-4 h-4" />,
    description: 'A large-scale model known for its reasoning capabilities.',
    speed: 2,
    intelligence: 5,
    context: '128K tokens',
    supports: {
      webSearch: true,
      vision: true,
      tools: true,
    },
    tags: [],
  },
];
